From Layers to Submodules: Rethinking Granularity in Replacement-Based LLM Compression
从层级到子模块:重思替换式LLM压缩的粒度
🧭 后训练压缩 · 子模块级残差替换 ★ 深度解读
University of Trento
提出子模块级非连续替换方法 SUBFIT,利用残差旁路拟合,在困惑度与准确率折衷上显著优于层级替换基线。
📌 工作参考价值(4 点)
- 对 LLM 部署中的推理加速有直接帮助:25% 稀疏度下可实现 1.12×~1.17× 的解码加速和 25% 的 KV 缓存节省,适合电商应用中需要低延迟、高并发的 agent 或推荐服务。
- 子模块级非连续选择策略可推广到其他模型的细粒度压缩,例如在生成式推荐或 agent 的多组件 Transformer 中,可独立对 Attention/FFN 进行按需裁剪,而非粗暴删除整个层。
- 闭式残差旁路拟合(利用特征协方差和低秩近似)仅需数千条校准样本,无需反向传播,这可以作为一种“剪枝后快速恢复”的手段嵌入到现有的 post-training 流水线中,降低压缩成本。
- 共享 FFN 输入基的设计在保持质量的同时大幅减少额外参数(约为原 FFN 参数的 1/10),类似 LoRA 的思路但应用于压缩,可参考其跨层参数共享方式以降低多任务或大模型的部署显存。